How to pass arguments in python functions